home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Extensions / Imaging / PIL / PpmImagePlugin.py < prev    next >
Encoding:
Text File  |  2000-06-23  |  2.7 KB  |  129 lines
  1. #
  2. # The Python Imaging Library.
  3. # $Id: PpmImagePlugin.py,v 1.1.1.1 1998/08/18 13:07:53 sjoerd Exp $
  4. #
  5. # PPM support for PIL
  6. #
  7. # History:
  8. #    96-03-24 fl    Created
  9. #    98-03-06 fl    Write RGBA images (as RGB, that is)
  10. #
  11. # Copyright (c) Secret Labs AB 1997-98.
  12. # Copyright (c) Fredrik Lundh 1996.
  13. #
  14. # See the README file for information on usage and redistribution.
  15. #
  16.  
  17.  
  18. __version__ = "0.2"
  19.  
  20. import string
  21.  
  22. import Image, ImageFile
  23.  
  24. #
  25. # --------------------------------------------------------------------
  26.  
  27. MODES = {
  28.     # standard
  29.     "P4": "1",
  30.     "P5": "L",
  31.     "P6": "RGB",
  32.     # extensions
  33.     "P0CMYK": "CMYK",
  34.     # PIL extensions (for test purposes only)
  35.     "PyP": "P",
  36.     "PyRGBA": "RGBA",
  37.     "PyCMYK": "CMYK"
  38. }
  39.  
  40. def _accept(prefix):
  41.     return prefix[0] == "P" and prefix[1] in "0456y"
  42.  
  43. class PpmImageFile(ImageFile.ImageFile):
  44.  
  45.     format = "PPM"
  46.     format_description = "Pbmplus image"
  47.  
  48.     def _token(self, s = ""):
  49.         while 1: # read until next whitespace
  50.         c = self.fp.read(1)
  51.         if not c or c in string.whitespace:
  52.             break
  53.         s = s + c
  54.     return s
  55.  
  56.     def _open(self):
  57.  
  58.     # check magic
  59.     s = self.fp.read(1)
  60.     if s != "P":
  61.         raise SyntaxError, "not a PPM file"
  62.     mode = MODES[self._token(s)]
  63.  
  64.     if mode == "1":
  65.         self.mode = "1"
  66.         rawmode = "1;I"
  67.     else:
  68.         self.mode = rawmode = mode
  69.  
  70.     for ix in range(3):
  71.         while 1:
  72.             while 1:
  73.             s = self.fp.read(1)
  74.             if s not in string.whitespace:
  75.             break
  76.         if s != "#":
  77.             break
  78.         s = self.fp.readline()
  79.         s = string.atoi(self._token(s))
  80.         if ix == 0:    
  81.         xsize = s
  82.         elif ix == 1:
  83.         ysize = s
  84.         if mode == "1":
  85.             break
  86.  
  87.     self.size = xsize, ysize
  88.     self.tile = [("raw",
  89.              (0, 0, xsize, ysize),
  90.              self.fp.tell(),
  91.              (rawmode, 0, 1))]
  92.  
  93.     # ALTERNATIVE: load via builtin debug function
  94.     # self.im = Image.core.open_ppm(self.filename)
  95.     # self.mode = self.im.mode
  96.     # self.size = self.im.size
  97.  
  98. #
  99. # --------------------------------------------------------------------
  100.  
  101. def _save(im, fp, filename):
  102.     if im.mode == "1":
  103.     rawmode, head = "1;I", "P4"
  104.     elif im.mode == "L":
  105.     rawmode, head = "L", "P5"
  106.     elif im.mode == "RGB":
  107.     rawmode, head = "RGB", "P6"
  108.     elif im.mode == "RGBA":
  109.     rawmode, head = "RGB", "P6"
  110.     else:
  111.     raise IOError, "cannot write mode %s as PPM" % im.mode
  112.     fp.write(head + "\n%d %d\n" % im.size)
  113.     if head != "P4":
  114.     fp.write("255\n")
  115.     ImageFile._save(im, fp, [("raw", (0,0)+im.size, 0, (rawmode, 0, 1))])
  116.  
  117.     # ALTERNATIVE: save via builtin debug function
  118.     # im._dump(filename)
  119.  
  120. #
  121. # --------------------------------------------------------------------
  122.  
  123. Image.register_open("PPM", PpmImageFile, _accept)
  124. Image.register_save("PPM", _save)
  125.  
  126. Image.register_extension("PPM", ".pbm")
  127. Image.register_extension("PPM", ".pgm")
  128. Image.register_extension("PPM", ".ppm")
  129.